package com.kaduoxee.ebooks.util;


import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

public class ImgVerifyCodeUtils {

    private final int width = 100;      //验证码图片的宽

    private final int height = 40;     //验证码图片的高

    private String text;                //用来保存验证码的文本内容

    private Random r = new Random();    //获取随机数对象

    private String[] fontNames = {"Georgia"};  //字体数组

    //验证码数组
    private String codes = "23456789abcdefghjkmnopqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ";

    /**
     * 获取随机颜色
     * @return 生成的颜色
     */
    private Color randomColor(){
        int r = this.r.nextInt(230);//这里为什么是230，因为当r，g，b都为255时，即为白色，为了好辨认，需要颜色深一点。
        int g = this.r.nextInt(230);
        int b = this.r.nextInt(230);
        return new Color(r, g, b);
    }

    /**
     *
     * @return 随机字体
     */
    private Font randomFont(){
        int index = r.nextInt(fontNames.length);
        String fontName = fontNames[index];
        int style = r.nextInt(4);   //随机获取字体的样式，0是无样式，1是加粗，2是斜体，3是加粗加斜体
        int size = r.nextInt(10)+24;//随机获取字体的大小
        return new Font(fontName, style, size);//返回生成的字体
    }

    /**
     *
     * @return 随机字符
     */
    private char randomChar(){
        int index = r.nextInt(codes.length());
        return codes.charAt(index);
    }

    /**
     * 画干扰线
     */
    private void drawLine(BufferedImage image){
        int num = r.nextInt(10);//定义干扰线数量
        Graphics2D g = (Graphics2D) image.getGraphics();
        for (int i = 0; i < num; i++) {
            int x1 = r.nextInt(width);
            int y1 = r.nextInt(height);
            int x2 = r.nextInt(width);
            int y2 = r.nextInt(height);
            g.setColor(randomColor());
            g.drawLine(x1, y1, x2, y2);
        }
    }

    /**
     *
     * 创建图片
     */
    private BufferedImage createImage(){
        //创建图片缓冲区
        BufferedImage image = new BufferedImage(width,height,BufferedImage.TYPE_INT_BGR);
        //获取画笔
        Graphics2D g = (Graphics2D) image.getGraphics();
        //设置背景色随机
//        g.setColor(new Color(255, 255, r.nextInt(245) + 10));
        g.setColor(Color.white);
        g.fillRect(0, 0, width, height);

        return image;
    }

    /**
     * 获取验证码图片的方法
     */
    public BufferedImage getImage() {
        BufferedImage image = createImage();
        Graphics2D g = (Graphics2D) image.getGraphics(); //获取画笔
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < 4; i++)             //画四个字符即可
        {
            String s = randomChar() + "";      //随机生成字符，因为只有画字符串的方法，没有画字符的方法，所以需要将字符变成字符串再画
            sb.append(s);                      //添加到StringBuilder里面
            float x = i * 1.0F * width / 4;   //定义字符的x坐标
            g.setFont(randomFont());           //设置字体，随机
            g.setColor(randomColor());         //设置颜色，随机
            g.drawString(s, x, height - 5);
        }
        this.text = sb.toString();
        drawLine(image);
        return image;
    }

    /**
     * 获取验证码文本的方法
     */
    public String getText() {
        return text;
    }


    /**
     * 输出画好的验证码
     */
    public static void output(BufferedImage image, OutputStream out) throws IOException {
        ImageIO.write(image, "JPEG", out);
    }


}
